from appium.webdriver.common.appiumby import AppiumBy
from app.wework.model.base_page import BasePage
from app.wework.pages.contacts_page import ContactsPage


class WeWorkApp(BasePage):
    def __init__(self, driver=None):
        super().__init__(driver)

    def contacts(self):
        """通讯录"""
        self.find(by=AppiumBy.CSS_SELECTOR, value='[text="通讯录"]').click()
        return ContactsPage(self.driver)

